# Description:
# TensorFlow Ranking Keras examples.

# Placeholder: load py_library
# Placeholder: load py_test
# Placeholder: load py_binary

package(
    default_visibility = [
        "//tensorflow_ranking:__subpackages__",
    ],
)

licenses(["notice"])

py_binary(
    name = "antique_kpl_din",
    srcs = ["antique_kpl_din.py"],
    python_version = "PY3",
    srcs_version = "PY3",
    deps = [
        # py/absl/flags dep,
        # py/tensorflow dep,
        "//tensorflow_ranking",
    ],
)

py_test(
    name = "antique_kpl_din_test",
    size = "large",
    srcs = ["antique_kpl_din_test.py"],
    python_version = "PY3",
    srcs_version = "PY3",
    tags = [
        "no_pip",
        "notsan",
    ],
    deps = [
        ":antique_kpl_din",
        # py/absl/testing:flagsaver dep,
        # Placeholder: proto upb dep
        # py/tensorflow dep,
        "//tensorflow_ranking",
        # tensorflow_serving/apis:input_proto_py_pb2 dep,
    ],
)

py_binary(
    name = "antique_ragged",
    srcs = ["antique_ragged.py"],
    python_version = "PY3",
    srcs_version = "PY3",
    deps = [
        # py/absl:app dep,
        # py/absl/flags dep,
        # py/absl/logging dep,
        # py/tensorflow dep,
        "//tensorflow_ranking",
    ],
)

py_test(
    name = "antique_ragged_test",
    size = "large",
    srcs = ["antique_ragged_test.py"],
    python_version = "PY3",
    srcs_version = "PY3",
    deps = [
        ":antique_ragged",
        # py/absl/testing:flagsaver dep,
        # py/absl/testing:parameterized dep,
        # Placeholder: proto upb dep
        # py/tensorflow dep,
        # tensorflow_serving/apis:input_proto_py_pb2 dep,
    ],
)

py_binary(
    name = "keras_dnn_tfrecord",
    srcs = ["keras_dnn_tfrecord.py"],
    python_version = "PY3",
    srcs_version = "PY3",
    deps = [
        # py/absl:app dep,
        # py/absl/flags dep,
        # py/tensorflow dep,
        "//tensorflow_ranking",
    ],
)

py_test(
    name = "keras_dnn_tfrecord_test",
    srcs = ["keras_dnn_tfrecord_test.py"],
    python_version = "PY3",
    srcs_version = "PY3",
    deps = [
        ":keras_dnn_tfrecord",
        # py/absl/testing:flagsaver dep,
        # py/absl/testing:parameterized dep,
        # Placeholder: proto upb dep
        # py/tensorflow dep,
        # tensorflow_serving/apis:input_proto_py_pb2 dep,
    ],
)

py_library(
    name = "tfrbert_task_experiments",
    srcs = ["tfrbert_task_experiments.py"],
    deps = [
        # tensorflow_models/official/core:config_definitions dep,
        # tensorflow_models/official/core:exp_factory dep,
        # tensorflow_models/official/modeling/optimization dep,
        "//tensorflow_ranking",
        "//tensorflow_ranking/extension/premade",
    ],
)

py_test(
    name = "tfrbert_task_experiments_test",
    srcs = ["tfrbert_task_experiments_test.py"],
    python_version = "PY3",
    srcs_version = "PY3",
    deps = [
        ":tfrbert_task_experiments",
        # Placeholder: proto upb dep
        # py/tensorflow dep,
        # tensorflow_models/official/core:config_definitions dep,
    ],
)

py_binary(
    name = "tfrbert_antique_train",
    srcs = ["tfrbert_antique_train.py"],
    python_version = "PY3",
    srcs_version = "PY3",
    deps = [
        ":tfrbert_task_experiments",
        # py/absl:app dep,
        # py/absl/flags dep,
        # py/gin dep,
        # py/tensorflow dep,
        # tensorflow_models/official/common:distribute_utils dep,
        # tensorflow_models/official/common:flags dep,
        # tensorflow_models/official/core:task_factory dep,
        # tensorflow_models/official/core:train_lib dep,
        # tensorflow_models/official/core:train_utils dep,
        # tensorflow_models/official/modeling:performance dep,
    ],
)
